while and for loops in python